import unittest

from quick_learn.chat_openai import ChatOpenAIHelper
from quick_learn.prompt_learn.chat_prompt_customize import ask1


class MyTestCase(unittest.TestCase):
    def test_something(self):
        self.assertEqual(True, False)  # add assertion here
    def test_ask_city_goods(self):
        chatHeler = ChatOpenAIHelper()
        chatHeler.ask_city_goods(city='广州')
    def test_ask_test(self):
        chatHeler = ChatOpenAIHelper()
        chatHeler.ask_test()
    def test_ask1(self):
        ask1()


if __name__ == '__main__':
    unittest.main()
